cd ../../

gpus=$1
ids_bsz=24

pretrained_prefix="['encoder_st.encoder.cnn.cnn.conv0',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm0',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg0',"` 
                  `"'encoder_st.encoder.cnn.cnn.conv1',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm1',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg1',"` 
                  `"'encoder_st.encoder.cnn.cnn.conv2',"` 
                  `"'encoder_st.encoder.cnn.cnn.batchnorm2',"` 
                  `"'encoder_st.encoder.cnn.cnn.cg2']"
desirable_prefix="['cnn.cnn.conv0',"` 
                `"'cnn.cnn.batchnorm0',"` 
                `"'cnn.cnn.cg0',"` 
                `"'cnn.cnn.conv1',"` 
                `"'cnn.cnn.batchnorm1',"` 
                `"'cnn.cnn.cg1',"` 
                `"'cnn.cnn.conv2',"` 
                `"'cnn.cnn.batchnorm2',"` 
                `"'cnn.cnn.cg2']"


echo "Run baseline with first 3-layer CNN initialized by Sup/SelfSL/Finetune results."
python sed_main.py   \
    --gpus "${gpus}," \
    --exp_name 'sed/init/F3C_semi_freeze_SUP/' \
    --batch_sizes "[0, 12, 12, 24, 0]" \
    --pretrained_ckpt "/home/shaonian/SED/sssl_sed/codes/exp/semi/BYOLA_3C+R_freeze_SUP/o192_i24/version_0/epoch=193-obj_metric=1.274.ckpt" \
    --pretrained_prefix ${pretrained_prefix} \
    --desirable_prefix ${desirable_prefix} \
    --freeze_cnn_layer "[0, 1, 2]"
